/*
 * Copyright 2025 coze-dev Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { type WriteableMessageLifeCycleServiceGenerator } from '@coze-common/chat-area';
import { Toast } from '@coze-arch/bot-semi';

import { catchParse } from '@/util';

import { type PluginBizContext } from '../../types/biz-context';

export const messageLifeCycleServiceGenerator: WriteableMessageLifeCycleServiceGenerator<
  PluginBizContext
> = plugin => ({
  onSendMessageError: ctx => {
    const { error } = ctx;
    if (error instanceof Error) {
      const res = catchParse<{ code?: number; msg?: string }>(
        error.message,
        {},
      );
      if (res?.code && res?.msg) {
        Toast.error(res.msg);
      }
    }
  },
  onBeforeSendMessage: ctx => {
    const { options = {} } = ctx;
    const optionNew = Object.assign({}, options, {
      extendFiled: {
        ...options?.extendFiled,
        extra: {
          ...(options?.extendFiled?.extra || {}),
          ...(plugin.pluginBizContext.extraBody || {}),
        },
      },
    });
    return { ...ctx, options: optionNew };
  },
});
